Trig graphs in python